from transformers import AutoModelForCausalLM, AutoTokenizer 
model_name = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B" 
tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(  model_name,
trust_remote_code=True,
torch_dtype=torch.bfloat16,
device_map="auto"
)
